# -*- coding: utf-8 -*-
# 导入相关的工具包
import torch
import torchvision.models as models
from torchsummary import summary

# 通过net.summay()查看网络的形状
# 模型实例化
model = models.vgg16()

summary(model=model, input_size=(3, 224, 224), batch_size=1, device='cpu')


















